cd ../..

gpus=$1

python -m trainer.IntplContrastiveTrainer test \
    --config configs/model/bigCRNN_contrastive.yaml \
    --config configs/dataset/desed_with_id.yaml \
    --config configs/trainer/default.yaml \
    --config configs/experiment/intplcontrastive.yaml \
    --model.exp_name="desed+CL/with_id/bigCRNN" \
    --data.init_args.batch_sizes=[12,12,12,192,96] \
    --ckpt_path=/home/shaonian/SED/SED/logs/2_baselines/bigCRNN/with_id/+data_augm/version_2/checkpoints/epoch106_metric0.1084.ckpt \
    --trainer.devices=${gpus} 